Step FunctionsにLambdaからデータを渡してみた

臼田佳祐

2017.06.27

この記事は公開されてから1年以上経過しています。情報が古い可能性がありますので、ご注意ください。

こんにちは、臼田です。

みなさんLambdaしてますか？

今日は、ちょっとしたwebページの更新チェックをしたくてLambdaを使おうと思ったのですが、せっかくなのでStep Functionsで分岐処理をしてみたいなーと思ったのでやってみました。

ちなみに、Step Functions初心者なので、これから始める人と同じ目線で説明していきます。

作りたいもの

最近脆弱性診断ができるAWS Inspectorを触っていたのですが、対応しているCVEのリストがこちらにあり、随時アップデートされています。

このファイルが更新されたことを通知する仕組みが作りたいと思って実装し始めました。

作りたいイメージ

だいたい下記のイメージです。※今後変更される可能性があります

Step FunctionsからTargetのUpdateを確認するLambdaを実行してレスポンスを取得
更新があればTweet用LambdaからTweet
途中でエラーが発生したらSESで通知

CVE_Check

数回のブログに分けて実装していきます。

今回の範囲

まずはUpdate Checkerをサクッと作って、それをStep Functionsから呼び出してレスポンスを取得してみます。

Webページの更新をチェックするLambda作成

それでは作っていきましょう。

中身のアルゴリズムとしては、該当URLに対してリクエストしてレスポンスヘッダーの「Last-Modified」をチェックして判断する感じにします。

Python 3.6で書いていきます。

コード

import urllib.request
from datetime import datetime, timedelta

def lambda_handler(event, context):
	url = "https://s3-us-west-2.amazonaws.com/rules-engine/CVEList.txt"
	yesterday = datetime.now() - timedelta(1)
	with urllib.request.urlopen(url) as r:
		lm_header = r.info()['Last-Modified']
		last_modified = datetime.strptime(lm_header, '%a, %d %b %Y %H:%M:%S %Z')
		print(last_modified)
	if last_modified > yesterday:
		result = {"update": True}
	else:
		result = {"update": False}
	return result

URLをハードコーディングしていますが、最初はとりあえずこんな感じでいいでしょう。

Step Functionsでつなぐので、レスポンスはjsonで返すようにします。

動きとしては、デイリーでLast-Modifiedをチェックして、1日以内の更新であれば「update: True」を返します。

本当はタイムゾーンとか気にするべきですが、ターゲットがGMTで返してくれるのと、pytzを利用しないと(色んな意味で)面倒になるので今回は無視します。

外部ライブラリを利用するなら、Serverless Frameworkを使ったほうがいいでしょう。

それでは、Lambdaを作成します。